home
***
CD-ROM
|
disk
|
FTP
|
other
***
search
/
Chip 2007 January, February, March & April
/
Chip-Cover-CD-2007-02.iso
/
Pakiet internetowy
/
Rozne
/
HTTrack 3.40-2
/
httrack-3.40-2.exe
/
{app}
/
src
/
mmsrip
/
mms.c
< prev
next >
Wrap
C/C++ Source or Header
|
2006-01-23
|
34KB
|
1,240 lines
/*
* $RCSfile: mms.c,v $
* $Date: 2006/01/23 20:30:43 $ - $Revision: 1.33 $
*
* This file is distributed as a part of MMSRIP ( MMS Ripper ).
* Copyright (c) 2005-2006 Nicolas BENOIT
*
* It is highly based on the work of SDP Multimedia and Major MMS.
* They deserve all the credits for it.
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2, or (at your option) any
* later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#define _GNU_SOURCE
#ifdef _WIN32
#include <winsock2.h>
#else
#include <unistd.h>
#include <strings.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#endif
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <errno.h>
#include "common.h"
#include "mms.h"
#include "error.h"
/*
* Put 32 bits inside a command buffer.
*/
static void
mms_put_32 ( MMS_PACKET *pak,
uint32_t value )
{
pak->buf[pak->num_bytes ] = value % 256;
value = value >> 8;
pak->buf[pak->num_bytes+1] = value % 256 ;
value = value >> 8;
pak->buf[pak->num_bytes+2] = value % 256 ;
value = value >> 8;
pak->buf[pak->num_bytes+3] = value % 256 ;
pak->num_bytes += 4;
return;
}
/*
* Returns the 32 bits located at specified offset.
*/
static uint32_t
mms_get_32 ( const unsigned char *buf,
const int offset )
{
return ( (((uint32_t)buf[offset+0]) << 0) |
(((uint32_t)buf[offset+1]) << 8) |
(((uint32_t)buf[offset+2]) << 16) |
(((uint32_t)buf[offset+3]) << 24) );
}
/*
* Returns the 64 bits located at specified offset.
*/
static uint64_t
mms_get_64 ( const unsigned char *buf,
const int offset )
{
return ( (((uint64_t)buf[offset+0]) << 0) |
(((uint64_t)buf[offset+1]) << 8) |
(((uint64_t)buf[offset+2]) << 16) |
(((uint64_t)buf[offset+3]) << 24) |
(((uint64_t)buf[offset+4]) << 32) |
(((uint64_t)buf[offset+5]) << 40) |
(((uint64_t)buf[offset+6]) << 48) |
(((uint64_t)buf[offset+7]) << 56) );
}
/*
* Converts a string into a UTF-16 string.
*/
static void
mms_string_utf16 ( uint8_t *dest,
unsigned char *src,
const ssize_t len )
{
ssize_t i;
memset ( dest, 0, len );
for ( i=0; i<len; ++i )
{
dest[i*2] = src[i];
dest[i*2+1] = 0;
}
dest[i*2] = 0;
dest[i*2+1] = 0;
return;
}
/*
* Checks a URL for a correct MMS protocol.
* Returns the host name index if the protocol is valid, else it returns MMS_RET_ERROR.
*/
static int
mms_check_protocol ( const char *url )
{
static const char *proto[] = { "mms://", "mmsu://", "mmst://", "mmsh://", NULL };
static const int proto_len[] = { 6, 7, 7, 7, 0 };
register const char *p = proto[0];
int i = 0;
while ( p != NULL )
{
if ( strncmp ( url, p, proto_len[i] ) == 0 )
return proto_len[i];
p = proto[++i];
}
return MMS_RET_ERROR;
}
/*
* Prints out a command packet (debug purpose).
*/
static void
mms_print_packet ( FILE *stddebug,
const MMS_PACKET *pak,
const ssize_t length,
const int orig )
{
ssize_t i;
unsigned char c;
fprintf ( stddebug, "\n-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-\n" );
if ( orig == MMS_SERVER )
fprintf ( stddebug, " command from server (%d bytes)\n", length );
else
fprintf ( stddebug, " command from client (%d bytes)\n", length );
fprintf ( stddebug, " start sequence: 0x%08x\n", mms_get_32(pak->buf, 0) );
fprintf ( stddebug, " command id: 0x%08x\n", mms_get_32(pak->buf, 4) );
fprintf ( stddebug, " length: 0x%8x \n", mms_get_32(pak->buf, 8) );
fprintf ( stddebug, " len8: 0x%8x \n", mms_get_32(pak->buf, 16) );
fprintf ( stddebug, " sequence #: 0x%08x\n", mms_get_32(pak->buf, 20) );
fprintf ( stddebug, " len8 (II): 0x%8x \n", mms_get_32(pak->buf, 32) );
fprintf ( stddebug, " dir | comm: 0x%08x\n", mms_get_32(pak->buf, 36) );
fprintf ( stddebug, " switches: 0x%08x\n", mms_get_32(pak->buf, 40) );
fprintf ( stddebug, "\nascii contents:\n" );
for ( i=48; i<length; i+=2 )
{
c = pak->buf[i];
if ( ( c>=32 ) && ( c<=128 ) )
fprintf ( stddebug, "%c", c );
else
fprintf ( stddebug, "." );
}
fprintf ( stddebug, "\n\npackage hexdump:\n " );
i = 0;
while ( 1 )
{
c = pak->buf[i];
fprintf ( stddebug, "%02x", c );
++i;
if ( i < length )
{
if ( (i % 16) == 0 )
fprintf ( stddebug, "\n" );
if ( (i % 2) == 0 )
fprintf ( stddebug, " " );
}
else
break;
}
fprintf ( stddebug, "\n\n" );
fflush ( stddebug );
return;
}
/*
* Sends a command packet to the server.
* Returns MMS_RET_SUCCESS if success, MMS_RET_ERROR either.
*/
static int
mms_send_packet ( MMS *mms,
const int command,
const uint32_t switches,
const uint32_t extra,
const ssize_t length,
const uint8_t *data )
{
MMS_PACKET pak;
ssize_t written_len;
const ssize_t len8 = ( length + (length%8) ) / 8;
pak.num_bytes = 0;
mms_put_32 ( &pak, 0x00000001 ); /* start sequence */
mms_put_32 ( &pak, 0xB00BFACE ); /* #-)) */
mms_put_32 ( &pak, length + 32 );
mms_put_32 ( &pak, 0x20534d4d ); /* protocol type "MMS " */
mms_put_32 ( &pak, len8 + 4 );
mms_put_32 ( &pak, mms->seq_num++ );
mms_put_32 ( &pak, 0x0 ); /* unknown */
mms_put_32 ( &pak, 0x0 );
mms_put_32 ( &pak, 2+len8 );
mms_put_32 ( &pak, 0x00030000 | command ); /* dir | command */
mms_put_32 ( &pak, switches );
mms_put_32 ( &pak, extra );
memcpy ( &pak.buf[48], data, length );
written_len = write ( mms->socket, pak.buf, length+48 );
if ( written_len == -1 )
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_send_packet", "write() said: %s", strerror(errno) );
#else
error ( "mms_send_packet", "write() failed" );
#endif
return MMS_RET_ERROR;
}
else if ( written_len != (length+48) )
{
if ( !mms->quiet )
error ( "mms_send_packet", "did not write enough bytes" );
return MMS_RET_ERROR;
}
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &pak, length+48, MMS_CLIENT );
return MMS_RET_SUCCESS;
}
/*
* Reads some bytes from the socket.
* This is blocking until we don't have the required amount of bytes.
* Returns MMS_RET_SUCCESS if success, MMS_RET_ERROR either.
*/
static int
mms_recv_packet ( const int s,
MMS_PACKET *pak,
size_t count,
const int quiet )
{
ssize_t read_len, total;
total = 0;
if ( MMS_BUF_SIZE < count )
{
if ( !quiet )
warning ( "mms_recv_packet", "caller is too greedy" );
count = MMS_BUF_SIZE;
}
while ( total < count )
{
read_len = read ( s, &pak->buf[total], count-total );
if ( read_len == -1 )
{
if ( !quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_recv_packet", "read() said: %s", strerror(errno) );
#else
error ( "mms_recv_packet", "read() failed" );
#endif
return MMS_RET_ERROR;
}
total += read_len;
}
return MMS_RET_SUCCESS;
}
/*
* Gets server's command.
* The first 8 bytes can be skipped setting offset to 8.
* It writes the packet length in the provided storage.
* Returns command ID, MMS_CMD_INVALID if error.
*/
static int
mms_recv_cmd_packet ( const int s,
MMS_PACKET *pak,
ssize_t *packet_len,
const int offset,
const int quiet )
{
MMS_PACKET tmp;
if ( ( offset > 8 ) || ( offset < 0 ) )
{
if ( !quiet )
error ( "mms_recv_cmd_packet", "provided offset is invalid" );
return MMS_RET_ERROR;
}
if ( mms_recv_packet ( s, &tmp, 12-offset, quiet ) != 0 )
{
if ( !quiet )
error ( "mms_recv_cmd_packet", "unable to get packet header" );
return MMS_RET_ERROR;
}
memcpy ( &pak->buf[offset], tmp.buf, 12-offset );
if ( offset == 0 )
{
if ( mms_get_32(pak->buf, 4) != 0xb00bface ) /* probably a coincidence... */
{
if ( !quiet )
error ( "mms_recv_cmd_packet", "answer should have been a cmd packet" );
return MMS_RET_ERROR;
}
}
memcpy ( packet_len, &pak->buf[8], 4 );
*packet_len = mms_get_32 ( (unsigned char *) packet_len, 0 ) + 4;
if ( ( *packet_len + 12 ) > MMS_BUF_SIZE )
{
if ( !quiet )
error ( "mms_recv_cmd_packet", "incoming packet is too big for me" );
return MMS_RET_ERROR;
}
if ( mms_recv_packet ( s, &tmp, *packet_len, quiet ) != 0 )
{
if ( !quiet )
error ( "mms_recv_cmd_packet", "unable to get packet body" );
return MMS_RET_ERROR;
}
memcpy ( &pak->buf[12], tmp.buf, *packet_len );
return ( mms_get_32(pak->buf, 36) & 0xFFFF );
}
/*
* Gets the header of the ASF/WMV stream/file.
* Returns the amount of bytes received, MMS_RET_ERROR if we encounter an error.
*/
static ssize_t
mms_recv_header_packet ( MMS *mms,
MMS_PACKET *pak )
{
MMS_PACKET pre_header;
MMS_PACKET tmp;
ssize_t header_len;
ssize_t packet_len;
header_len = 0;
while ( 1 )
{
if ( mms_recv_packet ( mms->socket, &pre_header, 8, mms->quiet ) != 0 )
{
if ( !mms->quiet )
error ( "mms_recv_header_packet", "unable to get pre-header" );
return MMS_RET_ERROR;
}
if ( pre_header.buf[4] == 0x02 )
{
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "\n-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-\n" );
fprintf ( mms->stddebug, " getting header packet from server\n\n" );
}
if ( mms->stddebug != NULL )
{
static int i; /* static, so we don't recreate it for each call */
for ( i=0; i<8; ++i )
fprintf ( mms->stddebug, "pre_header.buf[%d] = 0x%02x (%d)\n", i, pre_header.buf[i], pre_header.buf[i] );
}
packet_len = ( pre_header.buf[7] << 8 | pre_header.buf[6] ) - 8;
if ( mms->stddebug != NULL )
fprintf ( mms->stddebug, "\nASF Header Packet (%d bytes)\n", packet_len );
if ( mms_recv_packet ( mms->socket, &tmp, packet_len, mms->quiet ) != 0 )
{
if ( !mms->quiet )
error ( "mms_recv_header_packet", "unable to get header" );
return MMS_RET_ERROR;
}
if ( ( header_len + packet_len ) > MMS_BUF_SIZE )
{
if ( !mms->quiet )
error ( "mms_recv_header_packet", "total header length is too big for me" );
return MMS_RET_ERROR;
}
memcpy ( &pak->buf[header_len], tmp.buf, packet_len );
header_len += packet_len;
if ( mms->stddebug != NULL )
fprintf ( mms->stddebug, "\n" );
/* test if end of header is reached */
if ( ( pak->buf[header_len-1] == 1) &&
( pak->buf[header_len-2] == 1 ) )
return header_len;
}
else /* we might receive an ack or an error */
{
int command = mms_recv_cmd_packet ( mms->socket, &tmp, &packet_len, 8, mms->quiet );
if ( command == -1 )
{
if ( !mms->quiet )
error ( "mms_recv_header_packet", "unable to get cmd packet" );
return MMS_RET_ERROR;
}
memcpy ( tmp.buf, pre_header.buf, 8 );
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &tmp, packet_len, MMS_SERVER );
if ( command == MMS_CMD_PING )
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, tmp.buf );
else if ( command != MMS_CMD_HEADER_DATA )
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_recv_header_packet", "unknown command 0x%02x\n", command );
#else
error ( "mms_recv_header_packet", "unknown command\n" );
#endif
return MMS_RET_ERROR;
}
}
}
}
/*
* Interprates the header (media packet size, etc.).
* Returns the size of a media packet (should be != 0 ).
*/
static ssize_t
mms_interp_header ( MMS *mms,
const uint8_t *header,
const ssize_t header_len )
{
int i;
ssize_t packet_length = 0;
mms->expected_file_size = header_len;
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "\n-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-\n" );
fprintf ( mms->stddebug, " header interpretation\n\n" );
}
/* parse header */
mms->num_stream_ids = 0;
i = 30;
while ( i < header_len )
{
uint64_t guid_1, guid_2, length;
if ( mms->stddebug != NULL )
{
/* this is MS GUID format */
fprintf ( mms->stddebug, "{%02X%02X%02X%02X-%02X%02X-%02X%02X-%02X%02X-%02X%02X%02X%02X%02X%02X}\n",
header[i+3], header[i+2], header[i+1], header[i+0],
header[i+5], header[i+4], header[i+7], header[i+6],
header[i+8], header[i+9], header[i+10], header[i+11],
header[i+12], header[i+13], header[i+14], header[i+15] );
}
guid_2 = mms_get_64 ( header, i );
i += 8;
guid_1 = mms_get_64 ( header, i );
i += 8;
length = mms_get_64 ( header, i );
i += 8;
if ( ( guid_1 == 0x6553200cc000e48eULL ) && ( guid_2 == 0x11cfa9478cabdca1ULL ) )
{
packet_length = mms_get_32 ( header, i+92-24 );
if ( packet_length != mms_get_32(header, i+72) )
warning ( NULL, "size of media packets is not constant, it should not happen" );
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "File Properties Object (%lld bytes)\n", length );
fprintf ( mms->stddebug, " -> broadcast bit: %d\n", mms_get_32(header,i+64)&1 );
fprintf ( mms->stddebug, " -> min packet length: %d\n", packet_length );
fprintf ( mms->stddebug, " -> max packet length: %d\n", mms_get_32(header, i+72) );
fprintf ( mms->stddebug, " -> number of media packets: %d\n\n", mms_get_32(header, i+32) );
}
if ( mms_get_32(header,i+64)&1 )
{
if ( ( !mms->quiet ) && ( !mms->is_live ) )
warning ( NULL, "stream seems to be live, an error may occur" );
mms->is_live = MMS_LIVE;
}
}
else if ( ( guid_1 == 0x6553200cc000e68eULL ) && ( guid_2 == 0x11cfa9b7b7dc0791ULL ) )
{
int stream_id = header[i+48] | header[i+49] << 8;
if ( mms->num_stream_ids < 20 )
{
mms->stream_ids[mms->num_stream_ids] = stream_id;
++mms->num_stream_ids;
}
else
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
warning ( "mms_interp_header", "too many streams, stream \'%d\' skipped", stream_id );
#else
warning ( "mms_interp_header", "too many streams, skipping.." );
#endif
}
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "Stream Object (%lld bytes)\n", length );
fprintf ( mms->stddebug, " -> stream id: %d\n\n", stream_id );
}
}
else if ( ( guid_1 == 0x6cce6200aa00d9a6ULL ) && ( guid_2 == 0x11cf668e75b22636ULL ) )
{
mms->expected_file_size += ( length - 50 ); /* valid values are at least 50 bytes (why?) */
if ( mms->stddebug != NULL )
fprintf ( mms->stddebug, "Data Object (%lld bytes)\n\n", length );
}
else if ( mms->stddebug != NULL )
{
if ( ( guid_1 == 0x6cce6200aa00d9a6ULL ) && ( guid_2 == 0x11cf668e75b22630ULL ) )
fprintf ( mms->stddebug, "Header Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0xcb4903c9a000f489ULL ) && ( guid_2 == 0x11cfe5b133000890ULL) )
fprintf ( mms->stddebug, "Simple Index Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0xbe4903c9a0003490ULL ) && ( guid_2 == 0x11d135dad6e229d3ULL) )
fprintf ( mms->stddebug, "Index Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0x6553200cc000e38eULL ) && ( guid_2 == 0x11cfa92e5fbf03b5ULL) )
fprintf ( mms->stddebug, "Header Extension Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0x6cce6200aa00d9a6ULL ) && ( guid_2 == 0x11cf668e75b22633ULL) )
fprintf ( mms->stddebug, "Content Description Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0x50a85ec9a000f097ULL ) && ( guid_2 == 0x11d2e307d2d0a440ULL) )
fprintf ( mms->stddebug, "Extended Content Description Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0xf64803c9a000a4a3ULL ) && ( guid_2 == 0x11d0311d86d15240ULL) )
fprintf ( mms->stddebug, "Codec List Object (%lld bytes)\n\n", length );
else if ( ( guid_1 == 0xb2a2c9976000828dULL ) && ( guid_2 == 0x11d1468d7bf875ceULL) )
fprintf ( mms->stddebug, "Stream Bitrate Properties Object (%lld bytes)\n\n", length );
else
fprintf ( mms->stddebug, "Unknown Object (%lld bytes)\n\n", length );
}
i += length-24;
}
return packet_length;
}
/*
* Gets a media packet.
* Returns the amount of bytes in the media packet (0 if EOF), MMS_RET_ACKED if ACKed,
* MMS_RET_NO_AUTH if authorization failed, MMS_RET_ERROR either.
*/
static ssize_t
mms_recv_media_packet ( MMS *mms,
MMS_PACKET *pak )
{
MMS_PACKET pre_header;
ssize_t packet_len;
if ( mms_recv_packet ( mms->socket, &pre_header, 8, mms->quiet ) != 0 )
{
if ( !mms->quiet )
error ( "mms_recv_media_packet", "unable to get pre-header" );
return MMS_RET_ERROR;
}
if ( pre_header.buf[4] == 0x04 )
{
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "\n-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-\n" );
fprintf ( mms->stddebug, " getting media packet from server\n\n" );
}
if ( mms->stddebug != NULL )
{
static int i; /* static, so we don't recreate it for each call */
for ( i=0; i<8; i++ )
fprintf ( mms->stddebug, "pre_header[%d] = 0x%02x (%d)\n", i, pre_header.buf[i], pre_header.buf[i] );
}
packet_len = (pre_header.buf[7] << 8 | pre_header.buf[6]) - 8;
if ( mms->stddebug != NULL )
fprintf ( mms->stddebug, "\nASF Media Packet (%d bytes)\n", packet_len );
memset ( pak->buf, 0, mms->media_packet_len );
if ( mms_recv_packet (mms->socket, pak, packet_len, mms->quiet) != 0 )
{
if ( !mms->quiet )
error ( "mms_recv_media_packet", "unable to get media packet" );
return MMS_RET_ERROR;
}
}
else
{
int command = mms_recv_cmd_packet ( mms->socket, pak, &packet_len, 8, mms->quiet );
if ( command == MMS_RET_ERROR )
{
if ( !mms->quiet )
error ( "mms_recv_media_packet", "unable to get cmd packet" );
return MMS_RET_ERROR;
}
memcpy ( pak->buf, pre_header.buf, 8 );
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, pak, packet_len, MMS_SERVER );
if ( command == MMS_CMD_PING )
{
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, pak->buf );
return MMS_RET_ACKED;
}
else if ( command == MMS_CMD_END_OF_STREAM )
return 0;
else if ( command == MMS_CMD_STREAM_SELECT_ACK ) /* it happens sometimes */
return MMS_RET_ACKED;
else if ( command == MMS_CMD_READY_TO_STREAM ) /* it always happen before the first media packet */
{
/* this happens on some server for some unknown reason */
if ( mms_get_32(pak->buf, 40) == 0x80070005 )
{
if ( !mms->quiet )
error ( "mms_recv_media_packet", "streaming denied (read manpage & retry later)" );
return MMS_RET_NO_AUTH;
}
return MMS_RET_ACKED;
}
else
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_recv_media_packet", "unknown command 0x%02x\n", command );
#else
error ( "mms_recv_media_packet", "unknown command\n" );
#endif
return MMS_RET_ERROR;
}
}
if ( mms->stddebug != NULL )
fprintf ( mms->stddebug, "\n" );
return packet_len;
}
/*
* Creates An MMS Struct
* Returns an initialized MMS Struct if successful, NULL if an error occurs.
* url must start with mms:// or mmst:// (mmsu:// and mmsh:// are OK too, but we won't use UDP or do MMS over HTTP)
* trick must be MMS_TRICK_DISABLED or MMS_TRICK_ENABLED
*/
MMS *
mms_create ( const char *url,
FILE *out,
FILE *stddebug,
const int trick,
const int quiet )
{
MMS *mms;
register const char *sep;
register const int host_idx = mms_check_protocol ( url );
register const int url_len = strlen ( url );
if ( stddebug != NULL )
{
fprintf ( stddebug, "\n\n********************************************************************************\n\n" );
fprintf ( stddebug, "Url -> \'%s\'\n", url );
}
if ( host_idx == MMS_RET_ERROR )
{
if ( !quiet )
error ( "mms_create", "bad protocol (mms:// was expected)" );
return NULL;
}
sep = strchr ( &url[host_idx], '/' );
if ( sep == NULL )
{
if ( !quiet )
error ( "mms_create", "url seems to be malformed" );
return NULL;
}
if ( ( mms = ( MMS * ) malloc ( sizeof(MMS) ) ) == NULL )
{
if ( !quiet )
error ( "mms_create", "unable to allocate memory" );
return NULL;
}
mms->host = malloc ( ( sep - &url[host_idx] ) + 1 );
strncpy ( mms->host, &url[host_idx], (sep - &url[host_idx]) );
mms->host[sep-&url[host_idx]] = '\0'; /* strndup */
mms->path = strdup ( sep+1 );
mms->out = out;
mms->seq_num = 0;
mms->expected_file_size = 0;
mms->is_live = MMS_NO_LIVE;
/* we try to guess stream type with the filename extension */
if ( ( sep = strchr(sep,'?') ) != NULL )
{
if ( ( *(sep-4) == '.' ) && /* this will never segfault because */
( *(sep-3) == 'w' ) && /* the url is at least 6 chars long (mms://) */
( *(sep-2) == 'm' ) &&
( *(sep-1) == 'v' ) )
mms->stream_type = MMS_WMV;
else
mms->stream_type = MMS_ASF;
}
else
{
if ( ( url[url_len-4] == '.' ) &&
( url[url_len-3] == 'w' ) &&
( url[url_len-2] == 'm' ) &&
( url[url_len-1] == 'v' ) )
mms->stream_type = MMS_WMV;
else
mms->stream_type = MMS_ASF;
}
mms->stddebug = stddebug;
mms->quiet = quiet;
mms->trick = ( ((trick==MMS_TRICK_DISABLED)||(trick==MMS_TRICK_ENABLED)) ? trick : MMS_TRICK_DISABLED );
if ( mms->stddebug != NULL )
{
fprintf ( mms->stddebug, "Host -> \'%s\'\nPath -> \'%s\'\n", mms->host, mms->path );
fprintf ( mms->stddebug, "Stream type -> %s\n", (mms->stream_type==MMS_WMV)?"MMS_WMV":"MMS_ASF" );
}
return mms;
}
/*
* Establishes a connection with the MMS server.
* Returns MMS_RET_SUCCESS if success, MMS_RET_ERROR either.
*/
int
mms_connect ( MMS* mms )
{
struct sockaddr_in sa;
struct hostent *hp;
if ( mms == NULL )
return -1;
if ( ( hp = gethostbyname(mms->host) ) == NULL )
{
if ( !mms->quiet )
error ( "mms_connect", "unable to resolve host name" );
return MMS_RET_ERROR;
}
bcopy ( (char *) hp->h_addr, (char *) &sa.sin_addr, hp->h_length );
sa.sin_family = hp->h_addrtype;
sa.sin_port = htons ( 1755 );
if ( ( mms->socket = socket(hp->h_addrtype, SOCK_STREAM, 0) ) == -1 )
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_connect", "socket() said: %s", strerror(errno) );
#else
error ( "mms_connect", "socket() failed" );
#endif
return MMS_RET_ERROR;
}
if ( connect ( mms->socket, (struct sockaddr *) &sa, sizeof sa ) != 0 )
{
if ( !mms->quiet )
#ifdef HAVE_VSNPRINTF
error ( "mms_connect", "connect() said: %s", strerror(errno) );
#else
error ( "mms_connect", "connect failed" );
#endif
return MMS_RET_ERROR;
}
return MMS_RET_SUCCESS;
}
/*
* Handshake with the MMS server.
* ( we don't care about server's answers )
*/
int
mms_handshake ( MMS *mms )
{
MMS_PACKET pak;
int cmd = MMS_CMD_PING;
ssize_t packet_len;
char str [ 1024 ];
uint8_t data [ 2092 ];
if ( mms == NULL )
return MMS_RET_ERROR;
/* who we are */
memset ( data, 0, sizeof (data) );
if ( mms->trick == MMS_TRICK_DISABLED )
{
#ifdef HAVE_SNPRINTF
snprintf ( str, sizeof(str), "\034\003NSPlayer/7.0.0.1956; {3300AD50-2C39-46c0-AE0A-60181587CBA}; Host: %s", mms->host );
#else
if ( (strlen("\034\003NSPlayer/7.0.0.1956; {3300AD50-2C39-46c0-AE0A-60181587CBA}; Host: ")+strlen(mms->host)) < sizeof(str) )
sprintf ( str, "\034\003NSPlayer/7.0.0.1956; {3300AD50-2C39-46c0-AE0A-60181587CBA}; Host: %s", mms->host );
else
{
error ( "mms_handshake", "host name is too long" );
return MMS_RET_ERROR;
}
#endif
}
else
strcpy ( str, "\034\003NSPlayer/4.1.0.3928; {3300AD50-2C39-46c0-AE0A-60181587CBA}" );
mms_string_utf16 ( data, (unsigned char *) str, strlen(str)+2 );
mms_send_packet ( mms, MMS_CMD_HELLO, 0, 0x0004000b, strlen(str)*2 + 6, data );
cmd = MMS_CMD_PING;
while ( cmd == MMS_CMD_PING )
{
cmd = mms_recv_cmd_packet ( mms->socket, &pak, &packet_len, 0, mms->quiet );
if ( cmd == MMS_CMD_PING )
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, pak.buf );
}
if ( cmd == MMS_CMD_INVALID )
{
if ( !mms->quiet )
error ( "mms_handshake", "unable to get cmd packet" );
return MMS_RET_ERROR;
}
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &pak, packet_len, MMS_SERVER );
/* transport protocol selection */
memset ( &data, 0, sizeof(data) );
mms_string_utf16 ( &data[8], (unsigned char *) "\002\000\\\\192.168.0.129\\TCP\\1037\0000", 28 );
mms_send_packet ( mms, MMS_CMD_PROTOCOL_SELECT, 0, 0, 28*2+8, data );
cmd = MMS_CMD_PING;
while ( cmd == MMS_CMD_PING )
{
cmd = mms_recv_cmd_packet ( mms->socket, &pak, &packet_len, 0, mms->quiet );
if ( cmd == MMS_CMD_PING )
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, pak.buf );
}
if ( cmd == MMS_CMD_INVALID )
{
if ( !mms->quiet )
error ( "mms_handshake", "unable to get cmd packet" );
return MMS_RET_ERROR;
}
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &pak, packet_len, MMS_SERVER );
/* requested file */
mms_string_utf16 ( &data[8], (unsigned char *) mms->path, strlen(mms->path) );
memset ( data, 0, 8 );
mms_send_packet ( mms, MMS_CMD_FILE_REQUEST, 0, 0, strlen(mms->path)*2+12, data );
cmd = MMS_CMD_PING;
while ( cmd == MMS_CMD_PING )
{
cmd = mms_recv_cmd_packet ( mms->socket, &pak, &packet_len, 0, mms->quiet );
if ( cmd == MMS_CMD_PING )
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, pak.buf );
}
if ( cmd == MMS_CMD_INVALID )
{
if ( !mms->quiet )
error ( "mms_handshake", "unable to get cmd packet" );
return MMS_RET_ERROR;
}
if ( cmd == MMS_CMD_STREAM_INFOS )
{
if ( mms_get_32 ( pak.buf, 48 ) == -1 )
{
if ( !mms->quiet )
error ( "mms_handshake", "stream infos are not available" );
return MMS_RET_ERROR;
}
if ( ( mms_get_32(pak.buf,72) == 0 ) || ( mms_get_32(pak.buf,72) == 0xFFFFFFFF ) )
{
mms->is_live = MMS_LIVE;
if ( !mms->quiet )
warning ( NULL, "stream seems to be live, an error may occur" );
}
}
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &pak, packet_len, MMS_SERVER );
return MMS_RET_SUCCESS;
}
/*
* Gets ASF/WMV stream/file header
*/
ssize_t
mms_write_stream_header ( MMS *mms )
{
MMS_PACKET pak;
ssize_t len;
if ( mms == NULL )
return MMS_RET_ERROR;
/* give me the stream header please... */
memset ( pak.buf, 0, 40 );
pak.buf[32] = 2;
mms_send_packet ( mms, MMS_CMD_HEADER_REQUEST, 1, 0, 40, pak.buf );
len = mms_recv_header_packet ( mms, &pak );
if ( len == -1 )
{
if ( !mms->quiet )
error ( "mms_get_stream_header", "unable to get stream header\n" );
return MMS_RET_ERROR;
}
else if ( len == 0 )
{
if ( !mms->quiet )
error ( "mms_get_stream_header", "no stream header available, file may not exist" );
return MMS_RET_ERROR;
}
fwrite ( pak.buf, len, 1, mms->out );
mms->media_packet_len = mms_interp_header ( mms, pak.buf, len );
if ( mms->media_packet_len > MMS_BUF_SIZE )
{
if ( !mms->quiet )
error ( "mms_get_stream_header", "media packet length is too big for me" );
return MMS_RET_ERROR;
}
return len;
}
/*
* Tells the server we're ready to rip the stream from the beginning (don't know what happens if it's a live)
* Returns 0 if success, -1 either.
*/
int
mms_begin_rip ( MMS *mms )
{
int i;
uint8_t data [ 1024 ];
MMS_PACKET pak;
ssize_t packet_len;
int cmd;
if ( mms == NULL )
return MMS_RET_ERROR;
/* media stream selection */
memset ( data, 0, 40 );
for ( i=1; i<mms->num_stream_ids; ++i )
{
data [ (i-1) * 6 + 2 ] = 0xFF;
data [ (i-1) * 6 + 3 ] = 0xFF;
data [ (i-1) * 6 + 4 ] = mms->stream_ids[i];
data [ (i-1) * 6 + 5 ] = 0x00;
}
/* we add the final bytes of the stream selector, necessary for ASF streams only */
if ( mms->stream_type == MMS_ASF )
{
data [ (mms->num_stream_ids-1) * 6 + 0 ] = 0x00;
data [ (mms->num_stream_ids-1) * 6 + 1 ] = 0x00;
data [ (mms->num_stream_ids-1) * 6 + 2 ] = 0x00;
data [ (mms->num_stream_ids-1) * 6 + 3 ] = 0x20;
data [ (mms->num_stream_ids-1) * 6 + 4 ] = 0xac;
data [ (mms->num_stream_ids-1) * 6 + 5 ] = 0x40;
data [ (mms->num_stream_ids-1) * 6 + 6 ] = 0x02;
mms_send_packet ( mms, MMS_CMD_STREAM_SELECT, mms->num_stream_ids, 0xFFFF | mms->stream_ids[0] << 16, (mms->num_stream_ids-1)*6+2+4, data );
}
else
{
mms_send_packet ( mms, MMS_CMD_STREAM_SELECT, mms->num_stream_ids, 0xFFFF | mms->stream_ids[0] << 16, (mms->num_stream_ids-1)*6+2, data );
}
cmd = MMS_CMD_PING;
while ( cmd == MMS_CMD_PING )
{
cmd = mms_recv_cmd_packet ( mms->socket, &pak, &packet_len, 0, mms->quiet );
if ( cmd == MMS_CMD_PING )
mms_send_packet ( mms, MMS_CMD_PONG, 0, 0, 0, pak.buf );
}
if ( cmd == MMS_CMD_INVALID )
{
if ( !mms->quiet )
error ( "mms_begin_rip", "unable to get server\'s confirmation" );
return MMS_RET_ERROR;
}
if ( mms->stddebug != NULL )
mms_print_packet ( mms->stddebug, &pak, packet_len, MMS_SERVER );
/* ready... */
memset ( data, 0, 40 );
for ( i=8; i<16; ++i )
data[i] = 0xFF;
data[20] = 0x04;
mms_send_packet ( mms, MMS_CMD_START_PACKET, 1, 0xFFFF | mms->stream_ids[0] << 16, 24, data );
return MMS_RET_SUCCESS;
}
/*
* Writes a media packet.
* Returns the amount of bytes written, 0 if EOF, MMS_RET_NO_AUTH if streaming is denied, MMS_RET_ERROR either.
*/
ssize_t
mms_write_stream_data ( MMS *mms )
{
ssize_t len;
MMS_PACKET pak;
if ( mms == NULL )
return 0;
do
{
len = mms_recv_media_packet ( mms, &pak );
}
while ( len == MMS_RET_ACKED ); /* while we receive ACK packets */
if ( len == 0 )
return 0;
else if ( len == -1 )
{
if ( !mms->quiet )
error ( "mms_write_stream_data", "mms_recv_media_packet failed" );
return MMS_RET_ERROR;
}
else if ( len == MMS_RET_NO_AUTH )
{
if ( !mms->quiet )
error ( "mms_write_stream_data", "mms_recv_media_packet failed" );
return MMS_RET_NO_AUTH;
}
fwrite ( pak.buf, mms->media_packet_len, 1, mms->out ); /* we might have read less than the arbitrary media packet size */
return mms->media_packet_len;
}
/*
* Closes the connection.
*/
void
mms_disconnect ( MMS *mms )
{
uint8_t data [ 1024 ];
if ( mms == NULL )
return;
mms_send_packet ( mms, MMS_CMD_BYE_BYE, 0, 0, 0, data );
if ( ( shutdown(mms->socket, SHUT_RDWR) | close(mms->socket)) != 0 )
if ( !mms->quiet )
warning ( "mms_disconnect", "unable to close the socket properly" );
return;
}
/*
* Destroys An MMS Struct
*/
void
mms_destroy ( MMS *mms )
{
if ( mms == NULL )
return;
if ( mms->host != NULL )
free ( mms->host );
if ( mms->path != NULL )
free ( mms->path );
free ( mms );
return;
}